|
|
Bus Passenger Flow Forecast Based on Attention and Time-Sharing Graph Convolutional Network |
ZHANG Wei1,2, ZHU Fenghua2,3, CHEN Yuanyuan2, LÜ Yisheng2 |
1. School of Artificial Intelligence, University of Chinese Academy of Sciences, Beijing 100049 2. State Key Laboratory for Management and Control of Complex Systems, Institute of Automation, Chinese Academy of Sciences, Beijing 100190 3. Cloud Computing Center, Chinese Academy of Sciences, Dongguan 523808 |
|
|
Abstract 1cvarying system. Therefore, the spatiotemporal correlation between different bus lines can hardly be built effectively. To solve this problem, an attention and time-sharing graph convolution based long short-term memory network for bus passenger flow forecast is proposed. Firstly, temporal features of historical data are extracted by long short-term memory network(LSTM), and then they are weighted by a channel-wise attention module. A time-sharing graph convolution approach is utilized to analyze the spatial dependencies among bus lines. Different adjacent matrices are selected according to time intervals, and non-Euclidean pair-wise correlations are modeled via graph convolution. Finally, the final prediction result is obtained by integrating the extracted spatiotemporal features and vector representations of external factors, like weather and holiday information. Experiments on real bus passenger flow datasets indicate that the proposed model improves the prediction accuracy and learning speed evidently.
|
Received: 17 August 2020
|
|
Corresponding Authors:
ZHU Fenghua, Ph.D., associate professor. His research interests include artificial transportation system, parallel transportation system and intelligent control.
|
About author:: ZHANG Wei, master student. His research interests include traffic data mining and inte-lligent transportation system. CHEN Yuanyuan, Ph.D., assistant professor. His research interests include social transportation system, and data-driven traffic modeling and prediction. LÜ Yisheng, Ph.D., associate professor. His research interests include intelligent transportation system, artificial intelligence, and parallel traffic management and control system. |
|
|
|
[1] LÜ Y S, DUAN Y J, KANG W W, et al. Traffic Flow Prediction with Big Data: A Deep Learning Approach. IEEE Transactions on Intelligent Transportation Systems, 2015, 16(2): 865-873. [2] LÜ Y S, CHEN Y Y, LI L, et al. Generative Adversarial Networks for Parallel Transportation Systems. IEEE Intelligent Transportation Systems Magazine, 2018, 10(3): 4-10. [3] LIU L B, QIU Z L, LI G B, et al. Contextualized Spatial-Temporal Network for Taxi Origin-Destination Demand Prediction. IEEE Transactions on Intelligent Transportation Systems, 2019, 20(10): 3875-3887. [4] WU Y K, TAN H C, QIN L Q, et al. A Hybrid Deep Learning Based Traffic Flow Prediction Method and Its Understanding. Transportation Research Part C(Emerging Technologies), 2018, 90: 166-180. [5] ZHANG J B, ZHENG Y, QI D K. Deep Spatio-Temporal Residual Networks for Citywide Crowd Flows Prediction // Proc of the 31st AAAI Conference on Artificial Intelligence. Palo Alto, USA: AAAI Press, 2017: 1655-1661. [6] GONG M, FEI X, WANG Z H, et al. Sequential Framework for Short-Term Passenger Flow Prediction at Bus Stop. Transportation Research Record(Journal of the Transportation Research Board), 2014, 2417(7): 58-66. [7] LIU Y, LIU Z Y, JIA R. DeepPF: A Deep Learning Based Architecture for Metro Passenger Flow Prediction. Transportation Research Part C(Emerging Technologies), 2019, 101: 18-34. [8] LI J, PENG H, LIU L, et al. Graph CNNs for Urban Traffic Pa-ssenger Flows Prediction // Proc of the IEEE SmartWorld, Ubiquitous Intelligence and Computing, Advanced and Trusted Computing, Scalable Computing and Communications, Cloud and Big Data Computing, Internet of People and Smart City Innovation. Washington, USA: IEEE, 2018: 29-36. [9] ZHAO S Z, NI T H. Notice of Retraction: Prediction of Short-Term Bus Routes Passage Flow Based on Wavelet Analysis // Proc of the 2nd International Conference on Advanced Computer Control. Wa-shington, USA: IEEE, 2010: 49-52. [10] LI X S, CHEN Z Y, ZHU F H, et al. Short-Term Bus Passenger Flow Forecast Based on Deep Learning // Proc of the International Conference on Security, Pattern Analysis, and Cybernetics. Wa-shington, USA: IEEE, 2018: 372-376. [11] 李 梅.基于深度学习的短时公交客流预测研究.硕士学位论文.北京:北京交通大学, 2019. (LI M. Research on Short-Time Bus Passenger Flow Forecasting Based on Deep Learning. Master Dissertation. Beijing, China: Beijing Jiaotong University, 2019.) [12] GENG X, LI Y G, WANG L Y, et al. Spatiotemporal Multi-graph Convolution Network for Ride-Hailing Demand Forecasting // Proc of the AAAI Conference on Artificial Intelligence. Palo Alto, USA: AAAI Press, 2019: 3656-3663. [13] HOCHREITER S, SCHMIDHUBER J. Long Short-Term Memory. Neural Computation, 1997, 9(8): 1735-1780. [14] HE K M, ZHANG X Y, REN S Q, et al. Deep Residual Learning for Image Recognition // Proc of the IEEE Conference on Computer Vision and Pattern Recognition. Washington, USA: IEEE, 2016: 770-778. [15] HU J, SHEN L, ALBANIE S, et al. Squeeze-and-Excitation Networks. IEEE Transactions on Pattern Analysis and Machine Intelligence, 2017, 42(8): 2011-2023. [16] HAMMOND D K, VANDERGHEYNST P, GRIBONVAL R. Wave-lets on Graphs via Spectral Graph Theory. Applied and Computational Harmonic Analysis, 2011, 30(2): 129-150. [17] KIPF T N, WELLING M. Semi-Supervised Classification with Graph Convolutional Networks[C/OL]. [2020-07-15]. https://arxiv.org/pdf/1609.02907.pdf. [18] ZWILLINGER D, KOKOSKA S. CRC Standard Probability and Statistics Tables and Formulae. Boca Raton, USA: Chapman & Hall, 2000. [19] LI Z X, JANARDHANAN M N, TANG Q H, et al. Mathematical Model and Metaheuristics for Simultaneous Balancing and Sequencing of a Robotic Mixed-Model Assembly Line. Engineering Optimization, 2018, 50(5): 877-893. [20] LOSHCHILOV I, HUTTER F. SGDR: Stochastic Gradient Descent with Warm Restarts[C/OL]. [2020-07-15]. https://arxiv.org/pdf/1608.03983v5.pdf. [21] AHMED M S, COOK A R. Analysis of Freeway Traffic Time-Series Data by Using Box-Jenkins Techniques. Transportation Research Record, 1979, 722: 1-9. [22] SHI X J, CHEN Z R, WANG H, et al. Convolutional LSTM Network: A Machine Learning Approach for Precipitation Nowcasting // Proc of the 28th International Conference on Neural Information Processing Systems. New York, USA: ACM, 2015: 802-810. [23] DAI X Y, FU R, ZHAO E M, et al. DeepTrend 2.0: A Light-Weighted Multi-scale Traffic Prediction Model Using Detrending. Transportation Research Part C(Emerging Technologies), 2019, 103: 142-157. [24] LI Y G, YU R S, SHAHABI C, et al. Diffusion Convolutional Recurrent Neural Network: Data-Driven Traffic Forecasting[C/OL].[2020-07-15]. https://arxiv.org/pdf/1707.01926.pdf. [25] GUO S G, LIN Y F, FENG N, et al. Attention Based Spatial-Temporal Graph Convolutional Networks for Traffic Flow Forecasting // Proc of the AAAI Conference on Artificial Intelligence. Palo Alto, USA: AAAI Press, 2019: 922-929. |
|
|
|